'''
@Company: TWL
@Author: xue jian
@Email: xuejian@kanzhun.com
@Date: 2020-06-24 18:12:21
'''
if __name__ == "__main__":
    from batch_train import *

    '''
    用开聊数据训练的deepfm模型，用于接开聊实时流，与天级更新的达成模型融合，做实验看结果。
    ps用的offline-03机器，url为172.21.39.3，模型文件的名字为chat_deepfm_id，对应的port为1111。
    '''
    date_range = DateRange('2020-06-11', '2020-06-21')
    dates = date_range.dates

    ps_url = '172.21.39.3'
    port = 1111
    model_name = 'chat_deepfm_id'
    net_name = 'chat_deepfm_id.net'
    f_path = '/data3/training_data/arc_six/bossrec_chat_fid_flow_cut/'
    labels = '0,one'

    batch_train = BatchTrainJson(model_name, ps_url, 10, dates, f_path, net_name, labels)
    batch_train.train()
